from PIL import Image
import os
import shutil

# 图片压缩批处理


def compressImage(srcPath, dstPath):
    for filename in os.listdir(srcPath):
        # 如果不存在目的目录则创建一个，保持层级结构
        if not os.path.exists(dstPath):
            os.makedirs(dstPath)

        # 拼接完整的文件或文件夹路径
        srcFile = os.path.join(srcPath, filename)
        dstFile = os.path.join(dstPath, filename)

        # 如果是文件就处理
        if os.path.isfile(srcFile):
            try:
                # 打开原图片缩小后保存，可以用if srcFile.endswith(".jpg")或者split，splitext等函数等针对特定文件压缩
                sImg = Image.open(srcFile)
                w, h = sImg.size
                # 设置压缩尺寸和选项，注意尺寸要用括号
                dImg = sImg.resize((int(w/4), int(h/4)), Image.ANTIALIAS)
                # 也可以用srcFile原路径保存,或者更改后缀保存，save这个函数后面可以加压缩编码选项JPEG之类的
                dImg.save(dstFile)
                print(dstFile+" 成功！")
            except Exception:
                print(dstFile+"失败！")

        # 如果是文件夹就递归
        if os.path.isdir(srcFile):
            compressImage(srcFile, dstFile)


if __name__ == '__main__':
    # 遍历待加入图片
    path = os.walk("D:/myworkspace/dataset/img/b_img")
    for root, dirs, files in path:
        for f in files:
            shutil.move(os.path.join(root, f), os.path.join(
                'D:/myworkspace/dataset/img/finish', f))  # 移动文件

    # 遍历删除压缩图片
    path = os.walk("DD:/myworkspace/dataset/img/compress")
    for root, dirs, files in path:
        for f in files:
            os.remove(os.path.join(root, f))

    # 遍历压缩图片
    compressImage("D:/myworkspace/dataset/img/finish",
                  "D:/myworkspace/dataset/img/compress")
